import axios, { AxiosRequestConfig, AxiosResponse } from 'axios';
import { Message, Modal } from '@arco-design/web-vue';
import { useUserStore } from '@/store';
import { getToken } from '@/utils/auth';
import { MmHttpResponse } from '@/types/http';

if (import.meta.env.VITE_API_BASE_URL) {
  axios.defaults.baseURL = import.meta.env.VITE_API_BASE_URL;
}

axios.interceptors.request.use(
  (config: AxiosRequestConfig) => {
    // 源自 arco pro 自带拦截器让每一个请求都携带 token
    // 这里使用 JWT，验证自定义头信息 key
    // Authorization 是 JWT 自定义请求头的关键字
    // 根据实际情况自己修改
    const token = getToken();
    if (token) {
      if (!config.headers) {
        config.headers = {};
      }
      config.headers.Authorization = `Bearer ${token}`;
    }
    return config;
  },
  (error) => {
    // do something
    return Promise.reject(error);
  }
);

// add response interceptors
axios.interceptors.response.use(
  (response: AxiosResponse<MmHttpResponse>)=> {
    const res = response.data;
    // if the custom code is not 20000, it is judged as an error.
    if (res.code !== '20000') {
      Message.error({
        content: res.message || 'Error',
        duration: 5 * 1000,
      });
      // arco 原生代码（50008: 非法 token; 50012: 其他客户端登录; 50014: token 过期;）
      // 50008: Illegal token; 50012: Other clients logged in; 50014: Token expired;
      if (
        ['50008', '50012', '50014'].includes(res.code) &&
        response.config.url !== '/api/user/info'
      ) {
        Modal.error({
          title: '退出确认',
          content: '您已经退出系统，您可以选择停留在当前页面或者重新登录。',
          okText: '重新登录',
          async onOk() {
            const userStore = useUserStore();
            await userStore.logout();
            window.location.reload();
          },
        });
      }
      return Promise.reject(new Error(res.message || 'Error'));
    }
    return res;
  },
  (error) => {
    Message.error({
      content: error.msg || 'Request Error',
      duration: 5 * 1000,
    });
    return Promise.reject(error);
  }
);
